Nested for loop and Nested while in Python